﻿#include <opencv2/opencv.hpp>
#include <opencv2/core/ocl.hpp>

#include "Class/recognizer.h"
#include "Class/videoProcess.h"


using namespace cv;
using namespace std;

int main()
{
    // 使用 opencl 加速
    ocl::setUseOpenCL(true);

    // 读取视频并判断是否可以读取，并且初始化视频处理对象
    videoProcess cap("一种写着数字敲一下会闪的灯板.mp4");
    if (!cap.isOpened())
    {
        return -1;
    }
    Mat frame;
    recognizer recognizer;
    while (true)
    {
        cap >> frame;
        if (frame.empty())
        {
            return -1;
        }
        cap.setFrame(frame);
        cap.setSize(640, 480);

        // 预处理
        cap.preProcessHSVCanny(
            0 , 85,
            0 , 85,
            254 , 255);
        cap.showProcessedFrame();

        // 识别装甲
        recognizer.getStructure(cap.getProcessedFrame(), cap.getFrame());
        cap.showFrame();

        // 按'q'退出
        if (waitKey((int)1000/cap.get(CAP_PROP_FPS)) == 'q') {
            break;
        }

    }

}